# -*- coding: utf-8 -*-
# @Time    : 2021/7/7 14:46
# @Author  : LuoTianHang

# ####################config.py 说明##########################
# this script is the config script for the whole project

# 全局变量配置类
class GetConfig:
    def __init__(self):
        # region 项目配置基本信息
        self.project_name = "pose estimation"
        self.use_cuda = True
        self.seed = 123
        # endregion

        # region 数据集和dataloader配置基本信息
        self.train_files = "data/train2.json"
        self.test_files = "data/valid.json"
        self.val_num = 0.05
        self.image_size = 416  # 让图片输入分辨率为416 * 416，这一步的目的是为了让后面产生的卷积特征图宽高都为奇数，这样就可以产生一个center cell
        self.train_batch_size = 8
        self.test_batch_size = 8
        # endregion

        # region 模型初始化配置
        self.resume = True
        self.pretrained_weight = "./weight/best.pth"
        # endregion

        # region 模型训练过程配置
        self.lr = 1e-3
        self.epochs = 200
        # endregion
